package com.sosky;

import java.util.ArrayList;
import java.util.List;

/**
 * @author huanghuanlono
 * @date 2021/11/18 15:19
 * <p>
 * 931. 下降路径最小和
 * <p>
 * 给你一个 n x n 的 方形 整数数组 matrix ，请你找出并返回通过 matrix 的下降路径 的 最小和 。
 * 下降路径 可以从第一行中的任何元素开始，并从每一行中选择一个元素。在下一行选择的元素和当前行所选元素最多相隔一列
 * （即位于正下方或者沿对角线向左或者向右的第一个元素）。具体来说，位置 (row, col) 的下一个元素应当是
 * (row + 1, col - 1)、(row + 1, col) 或者 (row + 1, col + 1) 。
 * <p>
 * 示例 1：
 * 输入：matrix = [[2,1,3],[6,5,4],[7,8,9]]
 * 输出：13
 * 解释：下面是两条和最小的下降路径，用加粗+斜体标注：
 * [[2,1,3],      [[2,1,3],
 * [6,5,4],       [6,5,4],
 * [7,8,9]]       [7,8,9]]
 * <p>
 * 示例 2：
 * 输入：matrix = [[-19,57],[-40,-5]]
 * 输出：-59
 * 解释：下面是一条和最小的下降路径，用加粗+斜体标注：
 * [[-19,57],
 * [-40,-5]]
 * <p>
 * 示例 3：
 * 输入：matrix = [[-48]]
 * 输出：-48
 * <p>
 * 提示：
 * n == matrix.length
 * n == matrix[i].length
 * 1 <= n <= 100
 * -100 <= matrix[i][j] <= 100
 */
public class Solution931 {
    public static void main(String[] args) {
//        int[][] matrix1 = {{2, 1, 3}, {6, 5, 4}, {7, 8, 9}};
//        System.out.println("Down Path cost: " + minFallingPathSum(matrix1));
//        int[][] matrix2 = {{-19, 57}, {-40, -5}};
//        System.out.println("Down Path cost: " + minFallingPathSum(matrix2));
//        int[][] matrix3 = {{-48}};
//        System.out.println("Down Path cost: " + minFallingPathSum(matrix3));
        int[][] matrix4 = {{100, -42, -46, -41}, {31, 97, 10, -10}, {-58, -51, 82, 89}, {51, 81, 69, -51}};
        System.out.println("Down Path cost: " + minFallingPathSum(matrix4));
    }

    public static int minFallingPathSum(int[][] matrix) {
        List<Integer> path = new ArrayList<>();
        int preIndex = 0;
        for (int i = 0; i < matrix.length; i++) {
            int curIndex = (preIndex - 1) < 0 ? 0 : preIndex - 1;
            for (int j = preIndex; j < matrix[i].length; j++) {
                if (i == 0) {
                    if (matrix[i][j] < matrix[i][curIndex]) {
                        curIndex = j;
                    }
                } else {
                    if (j < (preIndex - 1) || j > (preIndex + 1)) {
                        continue;
                    } else {
                        if (matrix[i][j] < matrix[i][curIndex]) {
                            curIndex = j;
                        }
                    }
                }
            }
            path.add(matrix[i][curIndex]);
            preIndex = curIndex;
        }
        System.out.println("Down Path: " + path);
        return path.stream().mapToInt(Integer::intValue).sum();
    }
}
